
from ultralytics.utils.benchmarks import benchmark


def main(args):
    benchmark(model='yolov8n.pt', data='coco8.yaml', imgsz=640, half=False, device=0)


if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser()
    parser.add_argument('--device', type=str, default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
    parser.add_argument('--verbose', action='store_true')
    args = parser.parse_args()
    main(args)
